/***************************************************************************
 *   Copyright (C) 2009 by Zheng Yonglei
 *   zhengyonglei@gmail.com   *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

#ifndef TRAINER_H
#define TRAINER_H

#include <qwt_plot.h>
#include <qwt_plot_grid.h>
#include <qwt_plot_marker.h>
#include <qwt_interval_data.h>
#include <qwt_plot_curve.h>
#include <qwt_symbol.h>

#include <QtGui/QWidget>
#include "ui_Trainer.h"

#include "TrainerThread.h"

class Trainer : public QWidget
{
    Q_OBJECT

public:
    Trainer(QWidget *parent = 0);
    ~Trainer();

public slots:
    void startTraining();
    void stopTraining();
    void testANN();

    void testWordIDChanged(int);
    void simSizeChanged(int);
    void simAvg();

    void MSEUpated(int epoch, float mse);
    void statusChanged(const QString& msg);
    void updateWordSim(QString word, vector<int> intvVect, vector<float> simValVect);

private:
    void simSingleWord();

private:
    Ui::TrainerClass ui;

    TrainerThread mTrainerThread;

    QwtPlot mMSEPlot;
    QwtPlotCurve mMSEPlotCrv;
    QwtArray<double> mEpochArray;
	QwtArray<double> mMSEArray;

	QwtPlot mWordSimPlot;
	QwtPlotCurve mWordSimPlotCrv;
};

#endif // TRAINER_H
